package com.tbit.main.controller;

import com.tbit.main.common.JsonResult;
import com.tbit.main.pojo.BatteryModel;
import com.tbit.main.pojo.Result;
import com.tbit.main.service.BatteryModelService;
import com.tbit.main.util.Assert;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.List;

/**
 * 电池方案
 *
 * @author Leon
 * 2020年4月9日 下午8:39:03
 */
@RestController
@RequestMapping("/batteryModel")
public class BatteryModelController {
    @Autowired
    private BatteryModelService batteryModelService;

    /**
     * 根据平台账号id查询
     */
    @RequestMapping("/getByAccountId")
    public JsonResult getByAccountId(String token, Integer accountId) {
        Assert.notNull(accountId, "AccountUser.agentId.notNull");

        List<BatteryModel> batteryModels = batteryModelService.getBatteryModel(accountId);

        return JsonResult.succ(batteryModels);
    }


    /**
     * 获取区域下所有的电池方案
     * @param accountId 区域id
     * @return Result<List<BatteryModel>>
     */
    @RequestMapping("/getAllBatteryModel")
    public Result<List<BatteryModel>> getAllBatteryModel(Integer accountId) {
        Assert.notNull(accountId, "AccountUser.agentId.notNull");
        return batteryModelService.getAllBatteryModel(accountId);
    }


}